package com.wushijia.java8.stream;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.IntStream;

/**
 * @author Benjamin Winterberg
 */
public class Streams7 {

  static class Foo {

    String name;
    List<Bar> bars = new ArrayList<>();

    Foo(String name) {
      this.name = name;
    }
  }

  static class Bar {

    String name;

    Bar(String name) {
      this.name = name;
    }
  }

  public static void main(String[] args) {
//        test1();
    test2();
  }

  static void test2() {
    IntStream.range(1, 4)
        .mapToObj(num -> new Foo("Foo" + num))
        .peek(f -> IntStream.range(1, 4)
            .mapToObj(num -> new Bar("Bar" + num + " <- " + f.name))
            .forEach(f.bars::add))
        .flatMap(f -> f.bars.stream())
        .forEach(b -> System.out.println(b.name));
  }

  static void test1() {
    List<Foo> foos = new ArrayList<>();

    IntStream
        .range(1, 4)
        .forEach(num -> foos.add(new Foo("Foo" + num)));

    foos.forEach(f ->
        IntStream
            .range(1, 4)
            .forEach(num -> f.bars.add(new Bar("Bar" + num + " <- " + f.name))));

    foos.stream()
        .flatMap(f -> f.bars.stream())
        .forEach(b -> System.out.println(b.name));
  }

}